package com.lyl.pipeline.config;

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.io.ClassPathResource;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.data.redis.core.script.RedisScript;
import org.springframework.scripting.support.ResourceScriptSource;

/**
 * @author Mr.Luo
 * @date 2022-08-02 16:28
 * @desc lua脚本配置
 */
@Configuration
public class LuaConfig {

    /**
     * ip池限制器redis lua脚本
     * @return RedisScript
     */
    @Bean("ipPoolLimiterScript")
    public RedisScript<Boolean> ipPoolLimiterScript() {
        DefaultRedisScript<Boolean> redisScript = new DefaultRedisScript<>();
        redisScript.setScriptSource(new ResourceScriptSource(new ClassPathResource("lua/ip_pool_limiter.lua")));
        redisScript.setResultType(Boolean.class);
        return redisScript;
    }

    /**
     * ip限流 redis lua脚本
     * @return RedisScript
     */
    @Bean("ipRateLimiterScript")
    public RedisScript<Boolean> ipRateLimiterScript() {
        DefaultRedisScript<Boolean> redisScript = new DefaultRedisScript<>();
        redisScript.setScriptSource(new ResourceScriptSource(new ClassPathResource("lua/ip_rate_limiter.lua")));
        redisScript.setResultType(Boolean.class);
        return redisScript;
    }
}
